function with function as argument python